package org.basis.algorithm.sort;

import org.basis.algorithm.common.SortUtil;

/**
 * 选择排序: 从0~n位置上，每次循环选择一个最小的数进行交换
 *
 * @author Mr_wenpan@163.com 2021/12/02 15:39
 */
public class SelectionSort {

    public static void main(String[] args) {
//        int[] arr = {1, 2, 6, 3, 9, 0, 2};
        int[] arr = {5, 2, 3, 1};
//        proccess(arr);
        quickSort(arr);
        SortUtil.printArr(arr);
    }

    private static void proccess(int[] arr) {
        if (arr == null || arr.length < 2) {
            return;
        }
        for (int i = 0; i < arr.length; i++) {
            int minPosition = i;
            for (int j = i; j < arr.length; j++) {
                if (arr[minPosition] > arr[j]) {
                    minPosition = j;
                }
            }
            // 交换
            SortUtil.swap(i, minPosition, arr);
        }
    }

    /**
     * 快排
     */
    public static void quickSort(int[] arr) {
        if (arr == null || arr.length < 2) {
            return;
        }
        doQuickSort(arr, 0, arr.length - 1);
    }

    public static void doQuickSort(int[] arr, int left, int right) {
        // basecase
        if (left >= right) {
            return;
        }
        int[] par = partition(arr, left, right);
        // 左区间
        doQuickSort(arr, left, par[0] - 1);
        // 右区间
        doQuickSort(arr, par[1], right);
    }

//    public static int[] partition(int[] arr, int l, int r) {
//        int less = l - 1;
//        int more = r;
//        while (l < more) {
//            if (arr[l] < arr[r]) {
//                swap(arr, ++less, l++);
//            } else if (arr[l] > arr[r]) {
//                swap(arr, --more, l);
//            } else {
//                l++;
//            }
//        }
//        swap(arr, more, r);
//        return new int[]{less + 1, more};
//    }

    public static int[] partition(int[] arr, int left, int right) {
        int p = left - 1;
        int q = right;
        int index = left;
        while (index < q) {
            if (arr[index] < arr[right]) {
                swap(arr, index++, ++p);
            } else if (arr[index] > arr[right]) {
                swap(arr, index, --q);
            } else {
                // 相等则直接往前走
                index++;
            }
        }

        // 交换最后一个位置
        swap(arr, right, index);

        return new int[]{p + 1, q};
    }


    /**
     * 交换方法
     */
    public static void swap(int[] arr, int i, int j) {
        int temp = arr[i];
        arr[i] = arr[j];
        arr[j] = temp;
    }

}
